1392F - Omkar and Landslide - CodeForces Solution

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int main(){
    ios_base::sync_with_stdio(0), cin.tie(0);
    ll n;
    cin >> n;
    ll s = 0;
    for (ll x, i = 0; i < n; ++i) {
        cin >> x;
        s += x;
        s -= i;
    }
    ll base = s / n;
    ll xtra = s % n;
    for (ll i = 0; i < n; ++i) {
        ll h = base + i;
        if (xtra > 0) {
            ++h;
            --xtra;
        }
        cout << h << ' ';
    }
    cout << '\n';
    return 0;
}